package leetcode;

/**
 * @author caifangyi
 * @date 2022/7/28
 */

import java.util.Arrays;
import java.util.HashMap;
import java.util.Map;
import java.util.Random;
import java.util.stream.Collectors;

/**
 * 1331. 数组序号转换
 *
 * 给你一个整数数组arr ，请你将数组中的每个元素替换为它们排序后的序号。
 *
 * 序号代表了一个元素有多大。序号编号的规则如下：
 *
 * 序号从 1 开始编号。
 * 一个元素越大，那么序号越大。如果两个元素相等，那么它们的序号相同。
 * 每个数字的序号都应该尽可能地小。
 * 
 *
 * 示例 1：
 *
 * 输入：arr = [40,10,20,30]
 * 输出：[4,1,2,3]
 * 解释：40 是最大的元素。 10 是最小的元素。 20 是第二小的数字。 30 是第三小的数字。
 * 示例 2：
 *
 * 输入：arr = [100,100,100]
 * 输出：[1,1,1]
 * 解释：所有元素有相同的序号。
 * 示例 3：
 *
 * 输入：arr = [37,12,28,9,100,56,80,5,12]
 * 输出：[5,3,4,2,8,6,7,1,3]
 *
 * 来源：力扣（LeetCode）
 * 链接：https://leetcode.cn/problems/rank-transform-of-an-array
 * 著作权归领扣网络所有。商业转载请联系官方授权，非商业转载请注明出处。
 */
class Day1331 {

    public static void main(String[] args) {

        Solution solution = new Solution();
        int[] arr = {37,12,28,9,100,56,80,5,12};
        int[] ints = solution.arrayRankTransform(arr);
        System.out.println(Arrays.stream(ints).mapToObj(value -> String.valueOf(value)).collect(Collectors.joining(",")));
    }

    static class Solution {
        public int[] arrayRankTransform(int[] arr) {
            int[] clone = arr.clone();
            quickArr(clone);
            Map<Integer, Integer> map = new HashMap<>();
            int count=0;
            for (int i : clone) {
                if(!map.containsKey(i))map.put(i, ++count);
            }

            int ans[] = new int[arr.length];
            for (int i = 0; i < arr.length; i++) {
                ans[i] = map.get(arr[i]);
            }

            return ans;
        }

        //快排
        public void quickArr(int arr[]){
            quickRandom(arr, 0, arr.length - 1);
        }

        private void quickRandom(int[] arr, int l, int r) {
            if(l<r){
                int pos = quickPartition(arr, l , r);
                quickRandom(arr, l, pos -1);
                quickRandom(arr, pos+1, r);
            }
        }

        private int quickPartition(int[] arr, int l, int r) {
            int random = new Random().nextInt(r-l)+l;
            swap(arr, random, r);
            return partion(arr, l, r);
        }

        private int partion(int[] arr, int l, int r) {
            int pivot = arr[r];
            int i = l - 1;
            for (int j = l; j < r; j++) {
                if(arr[j] <= pivot){
                    i++;
                    swap(arr, i, j);
                }
            }
            swap(arr, i+1, r);
            return i+1;
        }

        private void swap(int[] arr, int i, int j) {
            if(i!=j){
                arr[i] = arr[i] ^ arr[j];
                arr[j] = arr[i] ^ arr[j];
                arr[i] = arr[i] ^ arr[j];
            }
        }
    }
}
